from __future__ import annotations

import datetime as dt
from typing import TYPE_CHECKING, Any, Literal, overload

import ibis
from typing_extensions import reveal_type

if TYPE_CHECKING:
    import ibis.expr.types as ir
    from typing_extensions import TypeAlias

    from narwhals.typing import PythonLiteral

Incomplete: TypeAlias = Any
"""Marker for upstream issues."""


@overload
def lit(value: Literal[False, True], dtype: None = ...) -> ir.BooleanScalar: ...
@overload
def lit(value: Literal[-1, 0, 1, 2, 3], dtype: None = ...) -> ir.IntegerScalar: ...
@overload
def lit(value: float, dtype: None = ...) -> ir.FloatingScalar | ir.IntegerScalar: ...
@overload
def lit(value: str, dtype: None = ...) -> ir.StringScalar: ...
@overload
def lit(value: PythonLiteral | ir.Value, dtype: None = ...) -> ir.Scalar: ...
@overload
def lit(value: Any, dtype: Any) -> Incomplete: ...
def lit(value: Any, dtype: Any | None = None) -> Incomplete:
    """Alias for `ibis.literal`."""
    literal: Incomplete = ibis.literal
    return literal(value, dtype)


@overload
def lit_intuitive(value: bool, dtype: None = ...) -> ir.BooleanScalar: ...  # noqa: FBT001
@overload
def lit_intuitive(value: int, dtype: None = ...) -> ir.IntegerScalar: ...
@overload
def lit_intuitive(value: float, dtype: None = ...) -> ir.FloatingScalar: ...
@overload
def lit_intuitive(value: str, dtype: None = ...) -> ir.StringScalar: ...
@overload
def lit_intuitive(value: PythonLiteral | ir.Value, dtype: None = ...) -> ir.Scalar: ...
@overload
def lit_intuitive(value: Any, dtype: Any) -> Incomplete: ...
def lit_intuitive(value: Any, dtype: Any | None = None) -> Incomplete:
    literal: Incomplete = ibis.literal
    return literal(value, dtype)


def here_we_go() -> None:  # noqa: PLR0914, PLR0915
    # ruff: noqa: E741

    a: Literal[False] = False
    b: Literal[True] = True
    c: Literal[False, True] = True
    d: bool = False
    e: Literal[0] = 0
    f: Literal[1] = 1
    g: Literal[-1, 0, 1, 2, 3] = -1
    h: int = 1
    i: int = 4
    j: float = 0.0
    k: str = "k"
    l: PythonLiteral = 1
    m: dt.date = dt.date(2000, 1, 1)
    n: Any = 0

    a_1 = lit(a)
    a_2 = lit_intuitive(a)
    reveal_type(a_1), reveal_type(a_2)

    b_1 = lit(b)
    b_2 = lit_intuitive(b)
    reveal_type(b_1), reveal_type(b_2)

    c_1 = lit(c)
    c_2 = lit_intuitive(c)
    reveal_type(c_1), reveal_type(c_2)

    d_1 = lit(d)
    d_2 = lit_intuitive(d)
    reveal_type(d_1), reveal_type(d_2)

    e_1 = lit(e)
    e_2 = lit_intuitive(e)
    reveal_type(e_1), reveal_type(e_2)

    f_1 = lit(f)
    f_2 = lit_intuitive(f)
    reveal_type(f_1), reveal_type(f_2)

    g_1 = lit(g)
    g_2 = lit_intuitive(g)
    reveal_type(g_1), reveal_type(g_2)

    h_1 = lit(h)
    h_2 = lit_intuitive(h)
    reveal_type(h_1), reveal_type(h_2)

    i_1 = lit(i)
    i_2 = lit_intuitive(i)
    reveal_type(i_1), reveal_type(i_2)

    j_1 = lit(j)
    j_2 = lit_intuitive(j)
    reveal_type(j_1), reveal_type(j_2)

    k_1 = lit(k)
    k_2 = lit_intuitive(k)
    reveal_type(k_1), reveal_type(k_2)

    l_1 = lit(l)
    l_2 = lit_intuitive(l)
    reveal_type(l_1), reveal_type(l_2)

    m_1 = lit(m)
    m_2 = lit_intuitive(m)
    reveal_type(m_1), reveal_type(m_2)

    n_1 = lit(n)
    n_2 = lit_intuitive(n)
    reveal_type(n_1), reveal_type(n_2)